'''主界面核心代码'''
from core import student_src, teacher_src, admin_src

func_menu = '''
========== 欢迎使用选课系统 ==========
        1.学生系统
        2.教师系统
        3.管理员系统
        q.退出输入【q】
====================================
'''

menu_dict = {
    '1': student_src.main,
    '2': teacher_src.main,
    '3': admin_src.main,
}


def main():
    while True:
        print(func_menu)
        func_choice = input("请输入功能ID:").strip()
        if func_choice.upper() == 'Q':
            print('【已退出】')
            break
        elif func_choice not in menu_dict:
            print('【功能ID有误】')
        else:
            menu_dict[func_choice]()
    print('欢迎您的下次使用')
